'''
    自助存取款系统
    1.用户登录信息、用户账户信息
    2.登录系统
    3.查询余额
    4.存钱
    5.取钱
    6.退出系统
    7.界面交互
'''
# 1.登录信息和账户信息
user_dict = {'zhangsan':'pw01','lisi':'pw02','wanger':'pw03'}
account_dict = {'zhangsan':1000000,'lisi':888,'wanger':6000}

# 2.登录系统
def loginBank(username,password):
    if username in user_dict:
        if user_dict[username] == password:
            print('登录成功！')
            return username
        else:
            print('密码错误，请重新输入！')
    else:
        print('用户不存在！')

# 3.查询余额
def getMoney(username):
    account = account_dict[username]
    print('{:>12}'.format('余额：'),account)
    return account
# 4.存钱
def saveMoney(username):
    money = int(input('请输入存入金额（数字）：'))
    sum_money = getMoney(username) + money
    account_dict[username] = sum_money
    print('{:>10}'.format('存入后余额：'),sum_money)

# 5.取钱
def drawMoney(username):
    money = int(input('请输入取出金额（数字）：'))
    sum_money = getMoney(username) - money
    account_dict[username] = sum_money
    print('{:>10}'.format('取出后余额：'),sum_money)

# 6.退出系统
def loginOut():
    pass

# 7.界面交互
# 用户输入用户名和密码
username = input('请输入用户名：')
password = input('请输入密码：')
# 定义当前登录用户
cur_user = loginBank(username,password)
while cur_user in user_dict:
    print('=' * 8, '自助存取款系统', '=' * 8)
    print('{:3}{:10}{:8}'.format('','1.查询余额', '2.存入金额'))
    print('{:3}{:10}{:8}'.format('','3.取出金额', '4.退出系统'))
    print('=' * 30)
    key = input('请输入对应的选择：')
    # 根据输入值执行对应操作
    if key == '1':
        print('*' * 8, '查询余额', '*' * 8)
        getMoney(cur_user)
        input('按回车键继续~')
    elif key == '2':
        print('*' * 8, '存入金额', '*' * 8)
        saveMoney(cur_user)
        input('按回车键继续~')
    elif key == '3':
        print('*' * 8, '取出金额', '*' * 8)
        drawMoney(cur_user)
        input('按回车键继续~')
    elif key == '4':
        loginOut()
        print('*' * 8, '退出系统', '*' * 8)
        break
    else:
        print('*' * 8, '操作无效', '*' * 8)
